package arrays;

import java.util.Arrays;

/**
 * @author pengfei.hpf
 * @date 2020/3/31
 * @verdion 1.0.0
 * 259. 较小的三数之和
 * 给定一个长度为 n 的整数数组和一个目标值 target，寻找能够使条件 nums[i] + nums[j] + nums[k] < target 成立的三元组  i, j, k 个数（0 <= i < j < k < n）。
 *
 * 示例：
 *
 * 输入: nums = [-2,0,1,3], target = 2
 * 输出: 2
 * 解释: 因为一共有两个三元组满足累加和小于 2:
 *      [-2,0,1]
 *      [-2,0,3]
 * 进阶：是否能在 O(n2) 的时间复杂度内解决？
 */
public class ThreeSumSmaller {
    public int threeSumSmaller(int[] nums, int target) {
        if(nums == null || nums.length < 3){
            return 0;
        }
        int count = 0;
        int i = 0;
        int n = nums.length;
        Arrays.sort(nums);
        while(i <= n - 3){
            int j = i + 1;
            int k = n - 1;
            if(nums[i] + nums[j] + nums[j + 1] >= target){
                break;
            }
            while(j < k){
                int sum = nums[i] + nums[j] + nums[k];
                if(sum < target){
                    count += (k - j);
                    j ++;
                } else {
                    k --;
                }
            }
            i++;
        }
        return count;
    }
}
